package com.muyu.redisdemo.common;

import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.alibaba.fastjson2.TypeReference;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * @ClassName: RedisMediator
 * @Author: Zirun.Xin
 * @Date: 2025/4/3 下午5:02
 * @Version: 1.0
 * @Description: redis中介，用于从redis中读取内容或者更新数据
 */
@Slf4j
@Component
public class RedisMediator {

    /**
     * redis字符串Template
     */
    @Resource
    private StringRedisTemplate stringRedisTemplate;

    /**
     * 读取配置文件，redis超时时间
     */
    @Value(value = "${spring.redis.jedis.pool.timeout}")
    private Integer timeOut;

    /**
     * 普通缓存放入
     *
     * @param key   键
     * @param value 值
     */
    public void set(String key, String value) {
        this.set(key, value, timeOut);
    }

    /**
     * 普通缓存放入并设置过期时间
     *
     * @param key   键
     * @param value 值 String
     * @param time  过期时间
     */
    public void set(String key, String value, long time) {
        stringRedisTemplate.opsForValue().set(key, value, time, TimeUnit.SECONDS);
    }

    /**
     * 普通缓存放入并设置过期时间
     *
     * @param key      键
     * @param value    值 String
     * @param time     过期时间
     * @param timeUnit 时间单位
     */
    public void set(String key, String value, long time, TimeUnit timeUnit) {
        stringRedisTemplate.opsForValue().set(key, value, time, timeUnit);
    }

    /**
     * 对象缓存放入
     *
     * @param key   键
     * @param value 值
     */
    public void set(String key, Object value) {
        this.set(key, JSONObject.toJSONString(value));
    }

    /**
     * 对象缓存放入并设置过期时间
     *
     * @param key   键
     * @param value 值
     * @param time  过期时间
     */
    public void set(String key, Object value, long time) {
        this.set(key, JSONObject.toJSONString(value), time);
    }

    /**
     * 普通缓存获取
     *
     * @param key 键
     * @return 缓存内容
     */
    public String get(String key) {
        this.checkKey(key);
        return stringRedisTemplate.opsForValue().get(key);
    }

    /**
     * 对象缓存获取
     *
     * @param key           键
     * @param typeReference
     * @param <T>
     * @return
     */
    public <T> T get(String key, TypeReference<T> typeReference) {
        if (typeReference == null) {
            throw new IllegalArgumentException("类型引用不能为空");
        }
        try {
            return JSON.parseObject(this.get(key), typeReference);
        } catch (Exception e) {
            throw new RuntimeException("反序列化失败，键: " + key + ", 类型: " + typeReference.getType(), e);
        }
    }

    /**
     * 根据规则获取多个key
     *
     * @param pattern 规则
     * @return key数据容器
     */
    public Set<String> getKeys(String pattern) {
        return stringRedisTemplate.keys(pattern);
    }

    /**
     * 获取缓存失效时间
     *
     * @param key 键
     * @return 缓存失效时间
     */
    public Long getExpire(String key) {
        return this.getExpire(key, TimeUnit.SECONDS);
    }

    /**
     * 获取缓存失效时间
     *
     * @param key      键
     * @param timeUnit 时间单位
     * @return 缓存失效时间
     */
    public Long getExpire(String key, TimeUnit timeUnit) {
        return stringRedisTemplate.getExpire(key, timeUnit);
    }

    /**
     * 判断key是否存在
     *
     * @param key 键
     * @return ture存在 false不存在
     */
    public boolean hasKey(String key) {
        this.checkKey(key);
        return Boolean.TRUE.equals(stringRedisTemplate.hasKey(key));
    }

    /**
     * 删除缓存
     *
     * @param key 键
     */
    public void del(String key) {
        this.checkKey(key);
        stringRedisTemplate.delete(key);
    }

    /**
     * 刷新缓存失效时间
     *
     * @param key 键
     * @return 是否刷新成功
     */
    public Boolean expire(String key) {
        return this.expire(key, timeOut);
    }

    /**
     * 指定缓存失效时间
     *
     * @param key  键
     * @param time 时间（秒）
     * @return 是否成功
     */
    public Boolean expire(String key, long time) {
        return stringRedisTemplate.expire(key, time, TimeUnit.SECONDS);
    }

    /**
     * 检查key是否为空
     *
     * @param key 键
     */
    private void checkKey(String key) {
        if (StringUtils.isBlank(key)) {
            throw new RuntimeException("redis key不能为空");
        }
    }
}
